/*
 * @(#)PortConfigHandler.java	1.0 Apr 17, 2008
 *
 *	The MIT License
 *
 *	Copyright (c) 2008 Malachi de AElfweald <malachid@gmail.com>
 *
 *	Permission is hereby granted, free of charge, to any person obtaining a copy
 *	of this software and associated documentation files (the "Software"), to deal
 *	in the Software without restriction, including without limitation the rights
 *	to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 *	copies of the Software, and to permit persons to whom the Software is
 *	furnished to do so, subject to the following conditions:
 *
 *	The above copyright notice and this permission notice shall be included in
 *	all copies or substantial portions of the Software.
 *
 *	THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 *	IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 *	FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 *	AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 *	LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 *	OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 *	THE SOFTWARE.
 */
package org.eoti.io.args4j;

import org.eoti.net.PortConfiguration;
import org.eoti.net.PortMap;
import static org.eoti.net.PortConfiguration.Protocol.tcp;
import org.kohsuke.args4j.spi.OptionHandler;
import org.kohsuke.args4j.spi.Setter;
import org.kohsuke.args4j.spi.Parameters;
import org.kohsuke.args4j.CmdLineParser;
import org.kohsuke.args4j.OptionDef;
import org.kohsuke.args4j.CmdLineException;

public class PortConfigHandler extends OptionHandler<PortConfiguration>
{
	public PortConfigHandler(CmdLineParser parser, OptionDef option, Setter<? super PortConfiguration> setter)
	{
		super(parser, option, setter);
	}

	protected PortConfiguration getWellKnownPort(String spec)
	{
		try{
			return PortMap.get(spec);
		} catch(Exception e){
			return null;
		}
	}

	protected PortConfiguration getByProtoAndPort(String spec)
	{
		try{
			int slashIndex = spec.indexOf("/");
			if(slashIndex == -1)
				return null;

			PortConfiguration.Protocol proto = PortConfiguration.Protocol.valueOf(spec.substring(0, slashIndex));
			int port = Integer.parseInt(spec.substring(slashIndex+1));
			return new PortConfiguration(proto, port);
		} catch(Exception e){
			return null;
		}
	}

	protected PortConfiguration getTCPbyPort(String spec)
	{
		try{
			return new PortConfiguration(tcp, Integer.parseInt(spec));
		} catch(Exception e){
			return null;
		}
	}

	@Override
	public int parseArguments(Parameters parameters)
			throws CmdLineException
	{
		String spec = parameters.getParameter(0);

		PortConfiguration config = getWellKnownPort(spec);

		if(config == null)
			config = getByProtoAndPort(spec);

		if(config == null)
			config = getTCPbyPort(spec);

		if(config == null)
			throw new CmdLineException("Incorrect format.");

		setter.addValue(config);

		return 1;
	}


	@Override
	public String getDefaultMetaVariable()
	{
		return "PortConfiguration";
	}
}
